"""
灰度图像直方图均衡化
"""
import cv2
import matplotlib.pyplot as plt

img = cv2.imread('../data/sunrise.jpg',0)
cv2.imshow('img',img)

# 直方图均衡化
img_equ = cv2.equalizeHist(img)
cv2.imshow('img_equ',img_equ)

# 直方图
plt.figure('Hist')
plt.subplot(1,2,1)
# img.ravel() 二维数组拉伸成一维数组
plt.hist(img.ravel(),bins=256,range=[0,256])

plt.subplot(1,2,2)
# img.ravel() 二维数组拉伸成一维数组
plt.hist(img_equ.ravel(),bins=256,range=[0,256])
plt.show()

cv2.waitKey()
cv2.destroyAllWindows()